iT邦幫忙

2023 iThome 鐵人賽

DAY 19
0
AI & Data

深度學習-TensorFlow系列 第 19

DAY19-TensorFlow常用模組(下)

  • 分享至 

  • xImage
  •  

5. tf.function : 圖執行模式
使用tf.function來實現圖執行模式 (TensorFlow 2 ),從而將模型轉換為易於部署且高效能的TensorFlow圖模型。

使用靜態編譯將函數內的代數轉換成計算圖,因此對函數內可使用的話句有一定限制 (只有支援Python語言的一個子集),且需要函數內的操作本身能夠被建構為計算圖。

!! 不是任何函數都可以被@tf.function修飾

get_concrete_function : 能夠直接獲得tf.function所產生的計算圖以進行進一步處理和應用
https://ithelp.ithome.com.tw/upload/images/20231004/20162982ZSv74X80QI.png

6. tf.TensorArray : TensorFlow動態陣列
將一系列張量以陣列的方式依序存放起來,以便進一步處理。

arr = tf.TensorArray(dtype, size, dynamic_size=False) :
宣告一個大小為size,類型為dtype的 TensorArray arr

其讀取和寫入的方法如下:
write(index, value) :value寫入陣列的第index個位置

read(index) : 讀取陣列的第index個值
https://ithelp.ithome.com.tw/upload/images/20231004/20162982gZiqHO8oPg.png

7. tf.config : GPU的使用與分配
tf.config.list_physical_devices : 可以獲得當前主機上某種特定計算設備類型(GPU、CPU)。

tf.config.experimential.set_memory_growth : 可以透過這方法將GPU的顯示卡記憶體儲存使用原則設為「僅在需要時申請顯示卡記憶體儲存空間」
https://ithelp.ithome.com.tw/upload/images/20231004/20162982Zj0eDynqbe.png
tf.config.LogicalDeviceConfiguration : 設置TensorFlow固定消耗GPU:0 的1GB顯示卡內存空間。


上一篇
DAY18-TensorFlow常用模組(中)
下一篇
DAY20-scope命名法
系列文
深度學習-TensorFlow30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言